import { RoleMessageType } from "../enum";
import { BaseMessage, BaseMessageConfig } from "./base";

export interface SystemMessageInfo extends Omit<BaseMessageConfig, "roleMessageType"> {
    roleMessageType?: RoleMessageType.AgentSystem;
}

/**
 * Agent 系统消息类
 */
export class SystemMessage extends BaseMessage {
    constructor(info: SystemMessageInfo) {
        const roleMessageType = RoleMessageType.AgentSystem;
        super({ ...info, roleMessageType });
    }
}
